import asyncio
from datetime import datetime

from tortoise import fields, models

from app.settings import settings


class BaseModel(models.Model):
    id = fields.BigIntField(pk=True, index=True)

    async def to_dict(self, m2m: bool = False, exclude_fields: list[str] | None = None):
        """
        将模型实例转换为字典表示
        
        参数:
        m2m: 是否包含多对多字段数据，默认为False
        exclude_fields: 需要排除的字段列表，为None时视为空列表
        
        返回:
        dict: 包含字段名和对应值的字典，datetime类型会被转为格式化字符串
        """
        # 初始化排除字段列表
        if exclude_fields is None:
            exclude_fields = []

        d = {}
        # 处理基础字段（非m2m字段）
        for field in self._meta.db_fields:
            if field not in exclude_fields:
                value = getattr(self, field)
                # 特殊处理datetime类型的字段
                if isinstance(value, datetime):
                    value = value.strftime(settings.DATETIME_FORMAT)
                d[field] = value

        # 处理多对多字段（当m2m=True时）
        if m2m:
            # 创建并行任务获取所有m2m字段数据
            tasks = [
                self.__fetch_m2m_field(field, exclude_fields)
                for field in self._meta.m2m_fields
                if field not in exclude_fields
            ]
            # 异步并行执行所有m2m字段获取任务
            results = await asyncio.gather(*tasks)
            
            # 将获取的m2m数据合并到结果字典
            for field, values in results:
                d[field] = values

        return d

    async def __fetch_m2m_field(self, field, exclude_fields):
        """
        异步获取多对多关系字段的格式化数据
        
        该方法获取指定多对多关系字段的所有关联对象数据，并排除指定字段，
        同时对datetime类型字段进行格式化处理。

        Args:
            field (str): 要获取的多对多关系字段名
            exclude_fields (set|list): 需要排除的字段名集合

        Returns:
            tuple: 包含两个元素的元组：
                - field (str): 原始字段名
                - formatted_values (list): 格式化后的关联对象数据列表，
                  每个元素为排除指定字段后的字典，其中datetime类型已转为字符串
        """
        # 异步获取关联对象的所有字段值（字典列表形式）
        values = await getattr(self, field).all().values()
        formatted_values = []

        # 遍历每个关联对象的数据字典
        for value in values:
            formatted_value = {}
            # 过滤需排除字段并处理特殊类型转换
            for k, v in value.items():
                if k not in exclude_fields:
                    # 将datetime对象转换为格式化的字符串
                    if isinstance(v, datetime):
                        formatted_value[k] = v.strftime(settings.DATETIME_FORMAT)
                    else:
                        formatted_value[k] = v
            formatted_values.append(formatted_value)

        return field, formatted_values

    class Meta:
        abstract = True


class UUIDModel:
    uuid = fields.UUIDField(unique=True, pk=False, index=True)


class TimestampMixin:
    created_at = fields.DatetimeField(auto_now_add=True, index=True)
    updated_at = fields.DatetimeField(auto_now=True, index=True)
